# 引入合适的模块
import numpy as np
import matplotlib.pyplot as plt

# 模拟 123*321 RGB图像掩模，四角都为10*10的空白区域
img1 = np.zeros((123, 321, 3), dtype=np.ubyte)
a = [255, 255, 255]
img1[:10, :10] = a  # 左上角10*10的区域
img1[:10, -10:] = [255, 255, 255]  # 右上角10*10的区域
img1[-10:, :10] = [255, 255, 255]  # 左下角10*10的区域
img1[-10:, -10:] = [255, 255, 255]  # 右下角10*10的区域

# 模拟 123*321 RGB图像掩模，四角都占10%的空白区域
img2 = np.zeros((123, 321, 3), dtype=np.ubyte)
h, w = img2.shape[:2]  # 获得高与宽
r = 0.1  # 要留白的比例
hr = int(h * r)  # %10的高对应的像素数
wr = int(w * r)  # %10的宽对应的像素数
img2[:hr, :wr] = [255, 255, 255]  # 左上角宽高3%的区域
img2[:hr, -wr:] = [255, 255, 255]  # 右上角宽高3%的区域
img2[-hr:, :wr] = [255, 255, 255]  # 坐下角宽高3%的区域
img2[-hr:, -wr:] = [255, 255, 255]  # 右下角宽高3%的区域

fg, ax = plt.subplots(2, 1)
ax[0].imshow(img1)
ax[0].set_title("4 10*10 corner")
ax[1].imshow(img2)
ax[1].set_title("4 10%*10% corner")

plt.tight_layout()
plt.savefig("out.png")
plt.show()
